class Model:
    def __init__(self, short_name:str, name: str, url: str, apikey: str):
        self.short_name = short_name
        self.name = name
        self.url = url
        self.apikey = apikey


support_models = [
    Model("qwen-plus", "qwen-plus", "https://dashscope.aliyuncs.com/compatible-mode/v1", "ALI_API_KEY"),
    Model("qwen3", "qwen3:8b", "http://localhost:11434/v1", "ALI_API_KEY"),
    Model("ds32", "deepseek-r1:32b", "http://localhost:11434/v1", "ALI_API_KEY"),
]

def get_model(name: str):
    for model in support_models:
        if model.name == name:
            return model
        if model.short_name == name:
            return model
    return None
